Node / ReticulumProjects / MeshChatX.git / files / vendor / lxmfy / tests / test_reliability.py

Displaying Raw • Download

T8b949e"""Reliability and crash recovery tests for LXMFy."""

Tff7b72import T7ee787os

Tff7b72import T7ee787pytest

Tff7b72import T7ee787time

Tff7b72from T7ee787lxmfy Tff7b72import Te6edf3BotConfigTb4b4b4, Te6edf3LXMFBot

Tff7b72from T7ee787unittestT7ee787.T7ee787mock Tff7b72import Te6edf3MagicMock

Tf0883e@pytestTff7b72.Td2a8ffmarkTff7b72.Td2a8ffreliability

Tff7b72class T56d364TestReliabilityTb4b4b4:

T8b949e"""Long-term reliability and crash recovery tests."""

Tff7b72def Td2a8fftest_state_persistence_after_crashTb4b4b4(Tff7b72selfTb4b4b4, Te6edf3test_config_dirTb4b4b4)Tb4b4b4:

T8b949e"""Simulate a crash and verify state recovery from storage."""

Te6edf3storage_path Tff7b72= Tffa657strTb4b4b4(Te6edf3test_config_dir Tff7b72/ Ta5d6ff"Ta5d6ffrecovery_storageTa5d6ff"Tb4b4b4)

T8b949e# 1. Setup bot and save some state

Te6edf3config1 Tff7b72= Te6edf3BotConfigTb4b4b4(

Te6edf3nameTff7b72=Ta5d6ff"Ta5d6ffPersistentBotTa5d6ff"Tb4b4b4, Te6edf3storage_pathTff7b72=Te6edf3storage_pathTb4b4b4, Te6edf3test_modeTff7b72=Tff7b72True

Tb4b4b4)

Te6edf3bot1 Tff7b72= Te6edf3LXMFBotTb4b4b4(Tff7b72*Tff7b72*Te6edf3config1Tff7b72.Td2a8ff__dict__Tb4b4b4)

Te6edf3bot1Tff7b72.Td2a8ffstorageTff7b72.Td2a8ffsetTb4b4b4(Ta5d6ff"Ta5d6ffimportant_stateTa5d6ff"Tb4b4b4, Tb4b4b4{Ta5d6ff"Ta5d6ffdataTa5d6ff"Tb4b4b4: T79c0ff123Tb4b4b4}Tb4b4b4)

Te6edf3bot1Tff7b72.Td2a8ffstorageTff7b72.Td2a8ffsetTb4b4b4(Ta5d6ff"Ta5d6ffuser_pref:abcTa5d6ff"Tb4b4b4, Ta5d6ff"Ta5d6ffvalue1Ta5d6ff"Tb4b4b4)

T8b949e# 2. Simulate "crash" by deleting the bot instance without graceful shutdown

Tff7b72del Te6edf3bot1

T8b949e# 3. Restart bot with same storage

Te6edf3config2 Tff7b72= Te6edf3BotConfigTb4b4b4(

Te6edf3nameTff7b72=Ta5d6ff"Ta5d6ffPersistentBotTa5d6ff"Tb4b4b4, Te6edf3storage_pathTff7b72=Te6edf3storage_pathTb4b4b4, Te6edf3test_modeTff7b72=Tff7b72True

Tb4b4b4)

Te6edf3bot2 Tff7b72= Te6edf3LXMFBotTb4b4b4(Tff7b72*Tff7b72*Te6edf3config2Tff7b72.Td2a8ff__dict__Tb4b4b4)

T8b949e# 4. Verify state is recovered

Tff7b72assert Te6edf3bot2Tff7b72.Td2a8ffstorageTff7b72.Td2a8ffgetTb4b4b4(Ta5d6ff"Ta5d6ffimportant_stateTa5d6ff"Tb4b4b4) Tff7b72== Tb4b4b4{Ta5d6ff"Ta5d6ffdataTa5d6ff"Tb4b4b4: T79c0ff123Tb4b4b4}

Tff7b72assert Te6edf3bot2Tff7b72.Td2a8ffstorageTff7b72.Td2a8ffgetTb4b4b4(Ta5d6ff"Ta5d6ffuser_pref:abcTa5d6ff"Tb4b4b4) Tff7b72== Ta5d6ff"Ta5d6ffvalue1Ta5d6ff"

Tff7b72def Td2a8fftest_corrupted_storage_recoveryTb4b4b4(Tff7b72selfTb4b4b4, Te6edf3test_config_dirTb4b4b4)Tb4b4b4:

T8b949e"""Verify that the bot can handle/recover from corrupted storage files."""

Te6edf3storage_path Tff7b72= Te6edf3test_config_dir Tff7b72/ Ta5d6ff"Ta5d6ffcorrupt_storageTa5d6ff"

Te6edf3storage_pathTff7b72.Td2a8ffmkdirTb4b4b4(Te6edf3parentsTff7b72=Tff7b72TrueTb4b4b4, Te6edf3exist_okTff7b72=Tff7b72TrueTb4b4b4)

T8b949e# Create a corrupted JSON file in storage

Te6edf3corrupt_file Tff7b72= Te6edf3storage_path Tff7b72/ Ta5d6ff"Ta5d6ffcorrupt_data.jsonTa5d6ff"

Tff7b72with Tffa657openTb4b4b4(Te6edf3corrupt_fileTb4b4b4, Ta5d6ff"Ta5d6ffwTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3fTb4b4b4:

Te6edf3fTff7b72.Td2a8ffwriteTb4b4b4(Ta5d6ff"Ta5d6ff{Ta5d6ff invalid json...Ta5d6ff"Tb4b4b4)

Te6edf3config Tff7b72= Te6edf3BotConfigTb4b4b4(

Te6edf3nameTff7b72=Ta5d6ff"Ta5d6ffCorruptBotTa5d6ff"Tb4b4b4, Te6edf3storage_pathTff7b72=Tffa657strTb4b4b4(Te6edf3storage_pathTb4b4b4)Tb4b4b4, Te6edf3test_modeTff7b72=Tff7b72True

Tb4b4b4)

T8b949e# Initialization should not crash despite corrupted file

Te6edf3bot Tff7b72= Te6edf3LXMFBotTb4b4b4(Tff7b72*Tff7b72*Te6edf3configTff7b72.Td2a8ff__dict__Tb4b4b4)

Tff7b72assert Te6edf3botTff7b72.Td2a8ffstorage Tff7b72is Tff7b72not Tff7b72None

T8b949e# Attempting to read corrupted data should return None or handle gracefully

T8b949e# Depending on how Storage is implemented

Te6edf3val Tff7b72= Te6edf3botTff7b72.Td2a8ffstorageTff7b72.Td2a8ffgetTb4b4b4(Ta5d6ff"Ta5d6ffcorrupt_dataTa5d6ff"Tb4b4b4)

Tff7b72assert Te6edf3val Tff7b72is Tff7b72None Tff7b72or Tffa657isinstanceTb4b4b4(Te6edf3valTb4b4b4, Tffa657dictTb4b4b4) T8b949e# Should not raise Exception

Tff7b72def Td2a8fftest_long_running_message_processingTb4b4b4(Tff7b72selfTb4b4b4, Te6edf3test_config_dirTb4b4b4)Tb4b4b4:

T8b949e"""Simulate a long-running bot processing messages continuously."""

Te6edf3config Tff7b72= Te6edf3BotConfigTb4b4b4(

Te6edf3nameTff7b72=Ta5d6ff"Ta5d6ffStableBotTa5d6ff"Tb4b4b4,

Te6edf3storage_pathTff7b72=Tffa657strTb4b4b4(Te6edf3test_config_dir Tff7b72/ Ta5d6ff"Ta5d6ffstable_storageTa5d6ff"Tb4b4b4)Tb4b4b4,

Te6edf3test_modeTff7b72=Tff7b72TrueTb4b4b4,

Tb4b4b4)

Te6edf3bot Tff7b72= Te6edf3LXMFBotTb4b4b4(Tff7b72*Tff7b72*Te6edf3configTff7b72.Td2a8ff__dict__Tb4b4b4)

Te6edf3botTff7b72.Td2a8ffrouter Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)

Te6edf3processed_count Tff7b72= T79c0ff0

Tff7b72def Td2a8ffhandlerTb4b4b4(Te6edf3senderTb4b4b4, Te6edf3msgTb4b4b4)Tb4b4b4:

Tff7b72nonlocal Te6edf3processed_count

Te6edf3processed_count Tff7b72+Tff7b72= T79c0ff1

Tff7b72return Tff7b72True

Te6edf3botTff7b72.Td2a8ffmessage_handlersTff7b72.Td2a8ffappendTb4b4b4(Te6edf3handlerTb4b4b4)

Tff7b72import T7ee787LXMF

T8b949e# Process messages in a loop for a short while

T8b949e# In a real reliability test this would run for hours

Te6edf3start_time Tff7b72= Te6edf3timeTff7b72.Td2a8fftimeTb4b4b4(Tb4b4b4)

Tff7b72while Te6edf3timeTff7b72.Td2a8fftimeTb4b4b4(Tb4b4b4) Tff7b72- Te6edf3start_time Tff7b72< T79c0ff2Tb4b4b4: T8b949e# 2 seconds of intense processing

Te6edf3mock_msg Tff7b72= Te6edf3MagicMockTb4b4b4(Te6edf3specTff7b72=Te6edf3LXMFTff7b72.Td2a8ffLXMessageTb4b4b4)

Te6edf3mock_msgTff7b72.Td2a8ffcontent Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6fftest contentTa5d6ff"

Te6edf3mock_msgTff7b72.Td2a8ffhash Tff7b72= Te6edf3osTff7b72.Td2a8ffurandomTb4b4b4(T79c0ff16Tb4b4b4)

Te6edf3mock_msgTff7b72.Td2a8ffsource_hash Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6ffsourceTa5d6ff"

Te6edf3mock_msgTff7b72.Td2a8ffdestination_hash Tff7b72= Tb4b4b4(

Te6edf3botTff7b72.Td2a8fflocalTff7b72.Td2a8ffhash Tff7b72if Tffa657hasattrTb4b4b4(Te6edf3botTff7b72.Td2a8fflocalTb4b4b4, Ta5d6ff"Ta5d6ffhashTa5d6ff"Tb4b4b4) Tff7b72else Ta5d6ffbTa5d6ff"Ta5d6fflocalTa5d6ff"

Tb4b4b4)

Te6edf3mock_msgTff7b72.Td2a8fffields Tff7b72= Tb4b4b4{Tb4b4b4}

Te6edf3mock_msgTff7b72.Td2a8ffsignature_validated Tff7b72= Tff7b72True

Te6edf3botTff7b72.Td2a8ff_message_receivedTb4b4b4(Te6edf3mock_msgTb4b4b4)

Tff7b72assert Te6edf3processed_count Tff7b72> T79c0ff0

Tffa657printTb4b4b4(Ta5d6fffTa5d6ff"Ta5d6ffProcessed Tffd700{Te6edf3processed_countTffd700}Ta5d6ff messages reliably.Ta5d6ff"Tb4b4b4)

Tff7b72def Td2a8fftest_concurrent_storage_accessTb4b4b4(Tff7b72selfTb4b4b4, Te6edf3test_config_dirTb4b4b4)Tb4b4b4:

T8b949e"""Test storage reliability under concurrent access from multiple threads."""

Tff7b72from T7ee787threading Tff7b72import Te6edf3Thread

Te6edf3config Tff7b72= Te6edf3BotConfigTb4b4b4(

Te6edf3nameTff7b72=Ta5d6ff"Ta5d6ffConcurrentBotTa5d6ff"Tb4b4b4,

Te6edf3storage_pathTff7b72=Tffa657strTb4b4b4(Te6edf3test_config_dir Tff7b72/ Ta5d6ff"Ta5d6ffconcurrent_storageTa5d6ff"Tb4b4b4)Tb4b4b4,

Te6edf3test_modeTff7b72=Tff7b72TrueTb4b4b4,

Tb4b4b4)

Te6edf3bot Tff7b72= Te6edf3LXMFBotTb4b4b4(Tff7b72*Tff7b72*Te6edf3configTff7b72.Td2a8ff__dict__Tb4b4b4)

Te6edf3errors Tff7b72= Tb4b4b4[Tb4b4b4]

Tff7b72def Td2a8ffworkerTb4b4b4(Te6edf3worker_idTb4b4b4)Tb4b4b4:

Tff7b72tryTb4b4b4:

Tff7b72for Te6edf3i Tff7b72in Tffa657rangeTb4b4b4(T79c0ff100Tb4b4b4)Tb4b4b4:

Te6edf3key Tff7b72= Ta5d6fffTa5d6ff"Ta5d6ffthread_Tffd700{Te6edf3worker_idTffd700}Ta5d6ff_key_Tffd700{Te6edf3iTffd700}Ta5d6ff"

Te6edf3botTff7b72.Td2a8ffstorageTff7b72.Td2a8ffsetTb4b4b4(Te6edf3keyTb4b4b4, Tb4b4b4{Ta5d6ff"Ta5d6ffvalTa5d6ff"Tb4b4b4: Te6edf3iTb4b4b4}Tb4b4b4)

Te6edf3val Tff7b72= Te6edf3botTff7b72.Td2a8ffstorageTff7b72.Td2a8ffgetTb4b4b4(Te6edf3keyTb4b4b4)

Tff7b72if Te6edf3valTb4b4b4[Ta5d6ff"Ta5d6ffvalTa5d6ff"Tb4b4b4] Tff7b72!= Te6edf3iTb4b4b4:

Te6edf3errorsTff7b72.Td2a8ffappendTb4b4b4(Ta5d6fffTa5d6ff"Ta5d6ffData mismatch in worker Tffd700{Te6edf3worker_idTffd700}Ta5d6ff"Tb4b4b4)

Tff7b72except Tf85149Exception Tff7b72as Te6edf3eTb4b4b4:

Te6edf3errorsTff7b72.Td2a8ffappendTb4b4b4(Ta5d6fffTa5d6ff"Ta5d6ffException in worker Tffd700{Te6edf3worker_idTffd700}Ta5d6ff: Tffd700{Te6edf3eTffd700}Ta5d6ff"Tb4b4b4)

Te6edf3threads Tff7b72= Tb4b4b4[Te6edf3ThreadTb4b4b4(Te6edf3targetTff7b72=Te6edf3workerTb4b4b4, Te6edf3argsTff7b72=Tb4b4b4(Te6edf3iTb4b4b4,Tb4b4b4)Tb4b4b4) Tff7b72for Te6edf3i Tff7b72in Tffa657rangeTb4b4b4(T79c0ff5Tb4b4b4)Tb4b4b4]

Tff7b72for Te6edf3t Tff7b72in Te6edf3threadsTb4b4b4:

Te6edf3tTff7b72.Td2a8ffstartTb4b4b4(Tb4b4b4)

Tff7b72for Te6edf3t Tff7b72in Te6edf3threadsTb4b4b4:

Te6edf3tTff7b72.Td2a8ffjoinTb4b4b4(Tb4b4b4)

Tff7b72assert Tff7b72not Te6edf3errorsTb4b4b4, Ta5d6fffTa5d6ff"Ta5d6ffConcurrent storage errors: Tffd700{Te6edf3errorsTffd700}Ta5d6ff"